# autor: Rui Rebola
# -*- coding: utf-8 -*-
import sqlite3
from function import *

'''
ligacao de base de dados
'''

connection = sqlite3.connect('IPC_POrtugal_1997_2013.db')
c = connection.cursor()

"""
criacao da tabela e
insercao de dados
"""

def create_table(dados):
	c.execute('DROP TABLE IF EXISTS dados')
	c.execute('CREATE TABLE dados (Ano int, IPC float, VarAn float, rMinM float, rMaxM float, PIB float, RendNac float, RendDisp float, Renume float)')
	c.executemany('INSERT INTO dados VALUES (?,?,?,?,?,?,?,?,?)' , dados)
	connection.commit()
	print "Base de Dadods Criada"
	connection.close()

'''
le informacao da base de dados e grava num array
'''	

def read_data(column, array, array2,anoI, anoF):
	'''
	column - str - nome da coluna a onde ir buscar os dados
	array - array - array para guardar os dados a pesquisar
	array2 - array - array para guardar os anos
	anoI - str - ano inicial
	anoF - str - ano final
	'''
	
	sql = 'Select ' + column +' from dados Where Ano BETWEEN ' + anoI +' and ' + anoF
	sql2 = 'Select Ano from dados Where Ano BETWEEN ' + anoI +' and ' + anoF + ' and ' + column + ' is not null'
	
	val = []
	val2 = []
	
	for row in c.execute(sql):
		val.append(row)
	
	for row in c.execute(sql2):
		val2.append(row)	

	for x in range (len(val)):
		if val[x][0] != '':
			array.append(val[x][0])
			array2.append(val2[x][0])
	

	
	
